package com.sheng.leetcode.year2023.month03.day10;

import org.junit.Test;

import java.util.HashMap;
import java.util.Map;

/**
 * @author liusheng
 * @date 2023/03/10
 * <p>
 * 1590. 使数组和能被 P 整除<p>
 * <p>
 * 给你一个正整数数组 nums，请你移除 最短 子数组（可以为 空），使得剩余元素的 和 能被 p 整除。 不允许 将整个数组都移除。<p>
 * 请你返回你需要移除的最短子数组的长度，如果无法满足题目要求，返回 -1 。<p>
 * 子数组 定义为原数组中连续的一组元素。<p>
 * <p>
 * 示例 1：<p>
 * 输入：nums = [3,1,4,2], p = 6<p>
 * 输出：1<p>
 * 解释：nums 中元素和为 10，不能被 p 整除。我们可以移除子数组 [4] ，剩余元素的和为 6 。<p>
 * <p>
 * 示例 2：<p>
 * 输入：nums = [6,3,5,2], p = 9<p>
 * 输出：2<p>
 * 解释：我们无法移除任何一个元素使得和被 9 整除，最优方案是移除子数组 [5,2] ，剩余元素为 [6,3]，和为 9 。<p>
 * <p>
 * 示例 3：<p>
 * 输入：nums = [1,2,3], p = 3<p>
 * 输出：0<p>
 * 解释：和恰好为 6 ，已经能被 3 整除了。所以我们不需要移除任何元素。<p>
 * <p>
 * 示例  4：<p>
 * 输入：nums = [1,2,3], p = 7<p>
 * 输出：-1<p>
 * 解释：没有任何方案使得移除子数组后剩余元素的和被 7 整除。<p>
 * <p>
 * 示例 5：<p>
 * 输入：nums = [1000000000,1000000000,1000000000], p = 3<p>
 * 输出：0<p>
 * <p>
 * 提示：<p>
 * 1 <= nums.length <= 10^5<p>
 * 1 <= nums[i] <= 10^9<p>
 * 1 <= p <= 10^9<p>
 * <p>
 * 来源：力扣（LeetCode）<p>
 * 链接：<a href="https://leetcode.cn/problems/make-sum-divisible-by-p">1590. 使数组和能被 P 整除</a><p>
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。<p>
 */
public class LeetCode1590 {

    @Test
    public void test01() {
//        int[] nums = {3, 1, 4, 2};
//        int p = 6;
//        int[] nums = {6, 3, 5, 2};
//        int p = 9;
//        int[] nums = {1, 2, 3};
//        int p = 3;
//        int[] nums = {1, 2, 3};
//        int p = 7;
        int[] nums = {1000000000, 1000000000, 1000000000};
        int p = 3;
        System.out.println(new Solution().minSubarray(nums, p));
    }
}

class Solution {
    public int minSubarray(int[] nums, int p) {
        int n = nums.length;
        int m = 0;
        // 防止超出整型的大小范围
        for (int num : nums) {
            m = (m + num) % p;
        }
        // m 为数组 nums 的总和除以 p 之后的余数
        if (m == 0) {
            return 0;
        }
        int ans = n;
        int cur = 0;
        // 每个前缀和模 p 之后最后一次出现的位置
        Map<Integer, Integer> last = new HashMap<>();
        last.put(0, -1);
        for (int i = 0; i < n; i++) {
            // cur 为当前前缀和模 p 之后得到的答案
            cur = (cur + nums[i]) % p;
            // 如果存在一个以 nums[i] 结尾的子数组，删除之后，剩余元素之和模 p 的值为 0
            // 需要找到一个前缀和模 p 的值为 target 的位置 j，使得 (target + m - cur) mod p = 0，此时说明 j + 1，i这段删除即可满足条件
            int target = (cur - m + p) % p;
            if (last.containsKey(target)) {
                ans = Math.min(ans, i - last.get(target));
            }
            last.put(cur, i);
        }
        return ans == n ? -1 : ans;
    }
}
